import os
import shutil

# 源文件夹路径（包含10个子文件夹）
source_dir = './task_envs_gru_'

# 目标文件夹路径
target_dir = './task_envs_gru_server'

# 创建目标文件夹（如果不存在的话）
if not os.path.exists(target_dir):
    os.makedirs(target_dir)

# 初始化文件编号
file_counter = 0

# 遍历每个子文件夹
for i in range(10):
    # 获取子文件夹的路径
    subfolder = os.path.join(source_dir, f'task_envs_gru_{i}')
    
    # 确保子文件夹存在
    if os.path.exists(subfolder):
        # 遍历子文件夹中的json文件
        for json_file in sorted(os.listdir(subfolder)):
            # 确保文件是json文件
            if json_file.endswith('.json'):
                # 构造原始文件路径
                original_file_path = os.path.join(subfolder, json_file)
                
                # 构造新的文件名，重新编号
                new_file_name = f'task_{file_counter}.json'
                new_file_path = os.path.join(target_dir, new_file_name)
                
                # 复制文件到目标文件夹并重命名
                shutil.copy2(original_file_path, new_file_path)
                
                # 更新文件计数器
                file_counter += 1

print(f"完成！共复制了 {file_counter} 个文件到 {target_dir}")